from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import JsonResponse


class LoginRequiredJSONMixin(LoginRequiredMixin):
    """自定义判断用户是否已经登录的扩展类：返回JSON"""

    # 父类已经完成是否登录的操作，子类只需要关心未登录作何反应
    def handle_no_permission(self):
        """直接响应JSON数据"""
        return JsonResponse({'code': 4003, 'errormsg': '用户未登录!!'})
